# -*- coding: utf-8 -*-
# Created by zhangyanqi on 2018/4/19

"""
实现 strStr() 函数。

给定一个 haystack 字符串和一个 needle 字符串，在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在，则返回  -1。

示例 1:

输入: haystack = "hello", needle = "ll"
输出: 2
示例 2:

输入: haystack = "aaaaa", needle = "bba"
输出: -1
说明:

当 needle 是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。

对于本题而言，当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
"""


class Solution:
    def strStr(self, haystack, needle):
        """
        :type haystack: str
        :type needle: str
        :rtype: int
        """
        if needle == "":
            return 0
        if haystack == "":
            return -1

        pos = -1
        hay_len = len(haystack)
        needle_index = 0
        needle_len = len(needle)
        for i in range(0, hay_len):
            if needle[needle_index] == haystack[i]:
                if needle_index == 0:
                    pos = i
            else:
                needle_index = 0
                pos = -1
                continue
            if needle_index == needle_len - 1:
                return pos
            needle_index += 1
        return pos


if __name__ == "__main__":
    s = Solution()
    str_str = s.strStr("aaaa", "aaa")
    print(str_str)

    s1 ="aaaa"
    s2 ="aaa"
    index = s1.index(s2)
    print(index)
